home *** CD-ROM | disk | FTP | other *** search
/ PC World Komputer 2010 April / PCWorld0410.iso / pluginy Firefox / 220 / 220.xpi / chrome / flashgot.jar / content / flashgot / HttpInterceptor.js < prev    next >
Encoding:
JavaScript  |  2010-01-24  |  7.5 KB  |  230 lines
  1. /***** BEGIN LICENSE BLOCK *****
  2.  
  3.     FlashGot - a Firefox extension for external download managers integration
  4.     Copyright (C) 2004-2009 Giorgio Maone - g.maone@informaction.com
  5.  
  6.     This program is free software; you can redistribute it and/or modify
  7.     it under the terms of the GNU General Public License as published by
  8.     the Free Software Foundation; either version 2 of the License, or
  9.     (at your option) any later version.
  10.  
  11.     This program is distributed in the hope that it will be useful,
  12.     but WITHOUT ANY WARRANTY; without even the implied warranty of
  13.     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14.     GNU General Public License for more details.
  15.  
  16.     You should have received a copy of the GNU General Public License
  17.     along with this program; if not, write to the Free Software
  18.     Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19.                              
  20. ***** END LICENSE BLOCK *****/
  21.  
  22. function HttpInterceptor() {
  23.   CC["@mozilla.org/uriloader;1"].getService(
  24.     CI.nsIURILoader).registerContentListener(this);
  25. }
  26.  
  27. HttpInterceptor.prototype = {
  28.   
  29.   autoStart: false,
  30.   interceptAll: true,
  31.   bypassAutoStart: false,
  32.   forceAutoStart: false,
  33.   
  34.   lastPost: null, // last uploadChannel
  35.    
  36.   QueryInterface: xpcom_generateQI([
  37.     CI.nsIURIContentListener,
  38.     CI.nsIObserver, 
  39.     CI.nsISupportsWeakReference,
  40.     CI.nsISupports
  41.   ]),
  42.   
  43.   
  44.   setup: function() { // profile initialization
  45.     this.autoStart = fg.getPref("autoStart", false);
  46.     this.interceptAll = fg.getPref("interceptAll", true);
  47.   },
  48.   
  49.   dispose: function() {
  50.     CC["@mozilla.org/uriloader;1"].getService(
  51.         CI.nsIURILoader).unRegisterContentListener(this);
  52.   },
  53.   
  54.   log: function(msg) {
  55.     fg.log(msg);
  56.   },
  57.   
  58.   _shouldIntercept: function(contentType) {
  59.     // dump("FG: _shouldIntercept("+contentType+")\n");
  60.     if (this.bypassAutoStart) return false;
  61.     
  62.     if (!(fg.DMS && fg.DMS.found)) return false;
  63.     if (this.forceAutoStart) return true;
  64.     
  65.     if (!this.autoStart) return false;
  66.     
  67.     if (this.interceptAll &&
  68.       !/\bxpinstall|text|xml|vnd\.mozilla|multipart\/x-mixed-replace\b/.test(contentType)) {
  69.       return true;
  70.     }
  71.  
  72.     if (contentType == "application/x-unknown-content-type" || /\b(?:xml|rss|javascript|json)\b/.test(contentType)) return false;
  73.     var ms = MediaSniffer.mimeService;
  74.     return fg.extensions.some(function(e) {
  75.       try { return contentType == ms.getTypeFromExtensions(e); } catch(ex) { return false; }
  76.     });
  77.   }
  78. , 
  79.   _willHandle: function(url, contentType) {
  80.     if (!/^(http|https|ftp|sftp|rtsp|mms):/i.test(url) ) {
  81.       if ((/^\s*javascript/i).test(url)) this.log("JavaScript url intercepted: "+url);
  82.       return false;
  83.     }
  84.     return true;
  85.   }
  86. ,
  87.   extractPostData: function(channel, res) {
  88.     res = res || {};
  89.     res.postData = res.postStream = null;
  90.     if (channel instanceof CI.nsIUploadChannel &&
  91.        channel.uploadStream instanceof CI.nsISeekableStream) {
  92.       this.log("Extracting post data...");
  93.       var sis;
  94.       try {
  95.         res.postStream = channel.uploadStream;
  96.         res.postStream.seek(0, 0);
  97.         sis = CC['@mozilla.org/scriptableinputstream;1'].createInstance(CI.nsIScriptableInputStream);
  98.         sis.init(res.postStream);
  99.         var postData  = sis.read(sis.available());
  100.         res.postStream.seek(0, 0);
  101.         
  102.         // buffered persistent copy 
  103.         res.postStream = CC["@mozilla.org/io/string-input-stream;1"].createInstance(CI.nsIStringInputStream);
  104.         res.postStream.setData(postData, postData.length);
  105.         if (res.postStream instanceof CI.nsISeekableStream) res.postStream.seek(0, 0);
  106.         
  107.         // remove headers
  108.         postData = postData.replace(/\s+$/,'').split(/[\r\n]+/)
  109.         res.postData = postData[postData.length - 1];
  110.       } catch(ex) {
  111.         this.log(ex.message);
  112.       } finally {
  113.         if (sis) sis.close();
  114.       }
  115.     }
  116.     return res;
  117.   },
  118.   /* nsIURIContentListener */
  119.   
  120.   canHandleContent: function(contentType, isContentPreferred, desiredContentType) {
  121.     // dump("FG: canHandleContent "+contentType+")\n");
  122.     return this._shouldIntercept(contentType);
  123.   }
  124. ,
  125.   lastRequest: null,
  126.   doContent: function(contentType, isContentPreferred, channel, contentHandler) {
  127.  
  128.     channel.QueryInterface(CI.nsIChannel);
  129.     const uri = channel.URI;
  130.     // dump("FG: doContent " +contentType + " " + uri.spec + "\n");
  131.     if (!this._willHandle(uri.spec, contentType)) {
  132.       throw new Error("FlashGot not interested in " + contentType + " from " + uri.spec);
  133.     }
  134.     
  135.     this.log("Intercepting download...");
  136.  
  137.     const pathParts=uri.path.split(/\//);
  138.     var links = [ {
  139.      href: channel.URI.spec, 
  140.      description: pathParts[pathParts.length-1],
  141.      noRedir: true
  142.     } ];
  143.     
  144.     
  145.     
  146.     if (channel instanceof CI.nsIHttpChannel) {
  147.       links.referrer = channel.referrer && channel.referrer.spec || "";
  148.       this.extractPostData(channel, links);
  149.     }
  150.     
  151.     try {
  152.         links.document = channel.notificationCallbacks.QueryInterface(
  153.             CI.nsIInterfaceRequestor).getInterface(
  154.             CI.nsIDOMWindow).document;
  155.         links.browserWindow = DOM.getChromeWindow(links.document.defaultView.top);
  156.         if (links.browserWindow.wrappedJSObject) links.browserWindow = links.browserWindow.wrappedJSObject;
  157.       } catch(e) {
  158.         this.log("Can't set referrer document for " + links[0].href + " from " + links.referrer);
  159.     }
  160.     
  161.     var firstAttempt;
  162.     if (contentHandler) {
  163.       this.lastRequest = null;
  164.       firstAttempt = true;
  165.       this.forceAutoStart = false;
  166.     } else {
  167.       var requestLines = [ channel.requestMethod, links[0].href, links.referrer || "", links.postData || ""].join("\n\n");
  168.       firstAttempt = this.lastRequest != requestLines;
  169.       this.lastRequest = requestLines;
  170.     }
  171.     
  172.     if (firstAttempt) {
  173.        var self = this;
  174.        fg.delayExec(function() {
  175.           self.forceAutoStart = false;
  176.           if(fg.download(links))
  177.             self.log("...interception done!");
  178.         }, 10);
  179.     } else {
  180.       // dump("Second attempt, skipping.\n");
  181.       this.lastRequest = null;
  182.       this.forceAutoStart = false;
  183.     }
  184.     
  185.     if (!channel.isPending()) { 
  186.       try {
  187.         channel.requestMethod = "HEAD";
  188.         channel.loadFlags = CI.nsIChannel.LOAD_RETARGETED | CI.nsIChannel.LOAD_RETARGETED_DOCUMENT_URI | CI.nsICachingChannel.LOAD_ONLY_FROM_CACHE;
  189.       } catch(e) {}
  190.     }
  191.     channel.cancel(NS_BINDING_ABORTED); 
  192.  
  193.     this.log("Original request cancelled.");
  194.     
  195.     return true;
  196.   },
  197.   contentHandler: {
  198.       onStartRequest: function(request, context) { 
  199.         throw "cancelled"; 
  200.       }, 
  201.       onStopRequest: function() {}, 
  202.       onDataAvailable: function() {}
  203.    }
  204. ,
  205.   isPreferred: function(contentType, desiredContentType) {
  206.     // dump("FG: isPreferred("+contentType+","+desiredContentType+")\n");
  207.     return this._shouldIntercept(contentType);
  208.   }
  209. ,
  210.   onStartURIOpen: function(uri) {
  211.     // dump("FG: onStartURIOpen "+ uri + (uri && uri.spec) + "\n");
  212.     return false;
  213.   }
  214. ,
  215.   // http-on-modify-request Observer 
  216.   observe: function(channel, topic, data) {
  217.     if (channel instanceof CI.nsIHttpChannel) {
  218.       
  219.       if (channel instanceof CI.nsIUploadChannel) {
  220.         this.lastPost = channel;
  221.       }
  222.       if (this.forceAutoStart) {
  223.         this.doContent("flashgot/forced", true, channel, null);
  224.         return;
  225.       }
  226.       
  227.       
  228.     }
  229.   }
  230. };